import requests
import json


def get_access_token():
    """
    使用 API Key，Secret Key 获取access_token，替换下列示例中的应用API Key、应用Secret Key
    """
    app_id = 88840877
    api_key = ""
    secret_key = ""
    url = (
        "https://aip.baidubce.com/oauth/2.0/token?grant_type=client_credentials&client_id=%s&client_secret=%s"
        % (api_key, secret_key)
    )

    payload = json.dumps("")
    headers = {"Content-Type": "application/json", "Accept": "application/json"}

    response = requests.request("POST", url, headers=headers, data=payload)
    return response.json().get("access_token")


def main():

    access_token = get_access_token()
    # print(access_token)
    url = (
        "https://aip.baidubce.com/rpc/2.0/ai_custom/v1/wenxinworkshop/chat/ernie-speed-128k?access_token="
        + access_token
    )
    message = {
        "messages": [
            {
                "role": "user",
                "content": "给你一个任务，对输入内容进行改写，改写成描述一副画的内容，改写的内容只能是对输入内容中已有的实物元素进行描述，或者添加一些简单的实实物元素，并且是比较精准的描写，描述的画面突出输入的内容，不要有情感、气氛、比喻，假想，联想等非实物的内容，不要漏掉输入内容中的实体元素。然后返回改写的内容给我，",
            },
            {"role": "assistant", "content": "好的，请输入你的内容。"},
            {"role": "user", "content": "一位漂亮的女孩\n"},
        ]
    }
    message = {
        "messages": [
            {"role": "user", "content": """
             我有一段文案，然后我要从文案中抽取一段内容来生成一张图片，并且这段内容生成的图片要贴合原文案相应的内容，

图片提示词内容大致为：(
人物描述：[具体的外貌特征，如帅气、阳光等]的[性别]孩，身着[服装款式和颜色]，[发型特征]
动作姿态：正在[具体的动作，如全力奔跑、慢跑等]，[身体部位的动作，如双臂摆动、双腿迈开等]
表情神态：[具体的神情，如专注、坚定、自信等]的表情，[面部细节，如汗水流淌、眉头紧皱等]
环境背景：在[具体的地点，如学校操场、公园跑道等]，[时间和天气，如夕阳西下、阳光明媚等]
其他细节：[其他需要突出的细节，如衣服湿透、鞋带松开等]。
)

以下为一个例子：【

[文案内容]：在空旷的操场上，有一个男孩正全力奔跑着。他身着简单的运动装，身影在夕阳的映照下显得格外矫健。
他的步伐轻盈而有力，每一次脚掌与地面的接触，都仿佛在释放着内心积压的能量。风掠过他的脸颊，吹乱了他的头发，却吹不散他专注的神情。
旁人的目光并未影响到他，他沉浸在自己的节奏中。呼吸略显急促，但那坚定的眼神表明他不会轻易停下。
他的双臂有规律地摆动着，像是在指挥着一场只属于他自己的交响乐。操场上的标线一格一格被他甩在身后，仿佛是他征服的领土。
这个男孩，或许是在为即将到来的比赛做准备，或许只是在享受奔跑带来的自由。但无论原因如何，他此刻的坚持和执着，都让这片操场充满了生机与活力。
汗水湿透了他的衣衫，可他浑然不觉，只是不断地向前、向前。在他的奔跑中，仿佛能看到对挑战的无畏，对目标的渴望。
夕阳渐渐西沉，将他的身影拉得更长，而他的脚步，依旧不停歇，仿佛要一直跑向那未知的远方。

[抽取的内容]：汗水湿透了他的衣衫，可他浑然不觉，只是不断地向前、向前。在他的奔跑中。

[图片提示词]：
人物描述：阳光帅气的男孩，身着白色运动 T 恤和黑色运动短裤，短发利落
动作姿态：正在全力奔跑，双臂有节奏地摆动
表情神态：神情专注，汗水顺着脸颊流淌
环境背景：在学校的操场上，夕阳的余晖映照
其他细节：衣服被汗水湿透

】。

[文案内容]:

杨安，十六岁，淬体境武者，清水县第一狗大户杨家家主嫡孙。

        号称清水县第一美男。

        但美中不足的是，他是传说中的低能儿。

        据说，其出生之时，天现异象，被雷劈坏了脑袋。

        此刻，因为强行突破淬骨境失败，才让杨安穿越而来。

        不是入赘，不是退婚。

        没有家族宗门被灭的苦大仇深。

        也不是被废的天才。

        随身老爷爷有木有？造化玉蝶啊随身空间啊有木有？

        或者外挂系统？

        杨安在脑海中绞尽脑汁嚎叫了大半天，毛的金手指都没发现。

        虽然有点点小小的失望，但杨安还能接受。

        甚至有点点兴奋。

        “不仅不瞎了，貌似也真的可以修炼了！还有，我的精神异变似乎依旧在呢，还有这这是天地灵气吗？是，绝对是！跟第二人生中的感觉一模一样！还有，我的身体貌似很强啊”

        没有悲伤，没有迷茫，不再惶恐！

        杨安清澈深邃的目光，闪耀着的是一抹兴奋和激动。

        这是一个跟第二人生相似的世界！

        他能清晰地感觉到，四周弥漫的天地灵气，以及此刻的肉身，赫然是第二人生中经历过的武道境界——淬体境巅峰！

        “砰！”

        感受着真实的力量，杨安全身肌肉筋脉鼓荡，气血凝聚，一拳轰在坚硬的岩石墙壁上。

        “真实的感觉这是真的！”

        深深地吸了口气，杨安听到门外的呼喊声，暂时压下心中的激动，起身，推开了修炼室的门。

        “小安！”

        门前，一名须发皆白的老者，神色期待地看着杨安。

        “爷爷？”

        脑海中的记忆，让杨安自然而然地叫出了声。

        眼前的老人正是杨家家主，杨朝舟，他的爷爷。

        “小安，成功没有？”

        杨朝舟问话的同时，直接上前一步抓住了杨安的手腕，顿时一股温热的能量便涌入杨安的体内。

        感觉像是蚂蚁沿着自己的经脉透过肌肉入侵他的四肢骨骼。

        杨安还没有回答，杨朝舟便是神色一暗，松开了手：

        “果然，破境丹都没用，唉”

        杨朝舟无奈叹息，旋即看向身后院门，挥手道：“都进来吧。”

        下一刻，杨安瞪大了眼睛，一脸懵圈。

        一名名打扮的花枝招展的少女鱼贯而入，或丰腴，或苗条，或玲珑娇俏，或高挑妖娆，大到十八九岁，小到十五六岁，一个个都含羞带涩地，来到杨安面前，列队而站，看着刚刚出关的杨安，欠身行礼：

        “拜见少爷。”

        “拜见少爷。”

        “拜见少爷”

        杨安看得眼花缭乱。

        “小安，这都是爷爷亲自帮你挑选的，我们清水县最漂亮且资质也不算差的女孩！你喜欢谁就挑出来，喜欢几个挑几个，全要都成。三天后，爷爷就给你举行清水县史上最隆重的婚礼！”

        我屮艸芔茻

        “爷爷，你这是要做什么？”

        杨安声音有点发颤，腿都在哆嗦。

        他的第四个梦想只是想要一个好看的小媳妇，可没想过这么多啊，更重要的是，这些女孩虽然也不难看，甚至有几个还是很漂亮的小姐姐，但明显没达到他的标准。

        再说了

        整整三十个！

        全要的话

        没有梨坏的田，只有累死的牛。

        老爷子这是想自己shi吗？

        “爷爷想抱孙子了。”

        杨朝舟看着自己的傻孙子，极为无奈地说道。

        “爷爷，我才十六岁啊”

        “小安啊，爷爷老了，没多久可活了。爷爷我含辛茹苦一把屎一把尿的将你拉扯大，可你都十六岁了，依旧无法突破到淬骨境，就连破境丹都失败了白云学府招生在即，你连报名的资格都没！唉，注定没希望了，爷爷只能寄希望到重孙儿身上了！就这么定了，三天后，爷爷给你举办婚礼！”

        老爷子霸气地说道。

        “爷爷，我拒绝！”

        “兔崽子，你想死？”杨朝舟吹胡子瞪眼。

        “不想。”

        “那你敢拒绝爷爷？爷爷为了我们这杨家容易吗我？你爹那个混蛋不孝子，一去十年不见踪影，连你也要反了？不听爷爷的话了是不是？”

        “听，我听啊，只是，爷爷”

        “没有只是，就这么定了！爷爷知道你喜欢婉儿，但人各有志，婉儿我们注定留不住的，你就不要想了！挑吧！”

[抽取内容]：
一名名打扮的花枝招展的少女鱼贯而入，或丰腴，或苗条，或玲珑娇俏，或高挑妖娆，大到十八九岁，小到十五六岁，一个个都含羞带涩地，来到杨安面前，列队而站，看着刚刚出关的杨安，欠身行礼：

请返回[图片提示词]。



             """},
        ]
    }
    payload = json.dumps(message)
    headers = {"Content-Type": "application/json"}

    response = requests.request("POST", url, headers=headers, data=payload)
    resp = json.loads(response.text)
    print(resp)


if __name__ == "__main__":
    main()
